// Copyright 2025 The Perses Authors
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

import { Query, QueryKey } from '@tanstack/react-query';
import { UnknownSpec, ProfileData, AbsoluteTimeRange } from '@perses-dev/core';
import { DatasourceStore } from '../runtime';
import { Plugin } from './plugin-base';

/**
 * A plugin for running profile queries.
 */
export interface ProfileQueryPlugin<Spec = UnknownSpec> extends Plugin<Spec> {
  getProfileData: (spec: Spec, ctx: ProfileQueryContext) => Promise<ProfileData>;
}

/**
 * Context available to ProfileQuery plugins at runtime.
 */
export interface ProfileQueryContext {
  datasourceStore: DatasourceStore;
  absoluteTimeRange?: AbsoluteTimeRange;
}

export type ProfileDataQuery = Query<ProfileData, unknown, ProfileData, QueryKey>;
